package com.bonc.apicreator.utils;


import java.io.BufferedInputStream;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


/**
 * 生成zip工具类
 *
 * @author gaopengyu
 * @version 2016年8月5日
 * @see ZipUtil
 * @since
 */
public final class ZipUtil {
    /**
     * 日志对象
     */
    private static final Logger LOGGER = LoggerFactory.getLogger(ZipUtil.class);

    private ZipUtil() {};

    /**
     * 创建ZIP文件
     * 
     * @param sourcePath
     *            文件或文件夹路径
     * @param zipPath
     *            生成的zip文件存在路径（包括文件名）
     */
    public static void createZip(String sourcePath, String zipPath) {
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        try {
            fos = new FileOutputStream(zipPath);
            zos = new ZipOutputStream(fos);
            writeZip(new File(sourcePath), "", zos);
        }
        catch (FileNotFoundException e) {
            LOGGER.error("File Not Found", e);
        }
        finally {
            try {
                if (zos != null) {
                    zos.close();
                }
            }
            catch (IOException e) {
                LOGGER.error("File Not Found", e);
            }

        }
    }

    /**
     * Description: 写zip
     * 
     * @param file
     * @param parentPath
     * @param zos
     *            void
     * @see
     */
    private static void writeZip(File file, String parentPath, ZipOutputStream zos) {
        if (file.exists()) {
            if (file.isDirectory()) {// 处理文件夹
                parentPath += file.getName() + File.separator;
                File[] files = file.listFiles();
                for (File f : files) {
                    writeZip(f, parentPath, zos);
                }
            }
            else {
                FileInputStream fis = null;
                DataInputStream dis = null;
                try {
                    fis = new FileInputStream(file);
                    dis = new DataInputStream(new BufferedInputStream(fis));
                    ZipEntry ze = new ZipEntry(parentPath + file.getName());
                    zos.putNextEntry(ze);
                    byte[] content = new byte[1024];
                    int len;
                    while ((len = fis.read(content)) != -1) {
                        zos.write(content, 0, len);
                        zos.flush();
                    }
                    fis.close();
                    dis.close();
                }
                catch (FileNotFoundException e) {
                    LOGGER.error("File Not Found", e);
                }
                catch (IOException e) {
                    LOGGER.error("File Not Found", e);
                }
            }
        }
    }
}